package com.team6.yygh.hosp.service.impl;

import com.alibaba.fastjson.JSONObject;
import com.team6.yygh.hosp.repository.DepartmentRepository;
import com.team6.yygh.hosp.service.DepartmentService;
import com.team6.yygh.model.hosp.Department;
import com.team6.yygh.vo.hosp.DepartmentQueryVo;
import com.team6.yygh.vo.hosp.DepartmentVo;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.*;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * @ClassName DepartmentServiceImpl
 * @Author ZhuYouWan
 * @Date 2022/10/20 9:40
 **/
@Service
public class DepartmentServiceImpl implements DepartmentService {
    @Autowired
    private DepartmentRepository departmentRepository;

    /**
     * 上传科室
     */
    @Override
    public void save(Map<String, Object> paramMap) {
        // paramMap 转换为 department 对象
        String paramMapString = JSONObject.toJSONString(paramMap);
        Department department = JSONObject.parseObject(paramMapString, Department.class);
        // 根据医院编号和科室编号查询是否存在
        Department departmentExist = departmentRepository.getDepartmentByHoscodeAndDepcode(department.getHoscode(), department.getDepcode());
        // 判断
        if (departmentExist != null) {
            departmentExist.setUpdateTime(new Date());
            departmentExist.setIsDeleted(0);
            departmentRepository.save(departmentExist);
        } else {
            department.setCreateTime(new Date());
            department.setUpdateTime(new Date());
            department.setIsDeleted(0);
            departmentRepository.save(department);
        }
    }

    @Override
    public Page<Department> findPageDepartment(int page, int limit, DepartmentQueryVo departmentQueryVo) {
        // 创建Pageable 对象， 设置当前页和每页记录数
        // 因为0是第一页，所以需要减一
        Pageable pageable = PageRequest.of(page-1, limit);
        // 创建Example对象
        Department department = new Department();
        BeanUtils.copyProperties(departmentQueryVo, department);
        department.setIsDeleted(0);
        ExampleMatcher matcher = ExampleMatcher.matching()
                .withStringMatcher(ExampleMatcher.StringMatcher.CONTAINING)
                .withIgnoreCase(true);
        Example<Department> example = Example.of(department, matcher);
        Page<Department> all = departmentRepository.findAll(example, pageable);
        return all;
    }

    @Override
    public void remove(String hoscode, String depcode) {
        // 根据医院编号 与 科室编号查询
        Department department = departmentRepository.getDepartmentByHoscodeAndDepcode(hoscode, depcode);
        if (department != null) {
            // 调用方法进行删除
            departmentRepository.deleteById(department.getId());
        }
    }

    /**
     * 根据医院编号，查询医院所有科室列表
     */
    @Override
    public List<DepartmentVo> findDeptTree(String hoscode) {
        // 创建list集合，用于最后封装
        List<DepartmentVo> result = new ArrayList<>();
        // 根据医院编号，查询医院所有科室信息
        Department departmentQuery = new Department();
        departmentQuery.setHoscode(hoscode);
        Example example = Example.of(departmentQuery);
        // 所有科室信息,包括大科室和小科室 departmentList
        List<Department> departmentList = departmentRepository.findAll(example);
        // 根据大科室编号 bigcode 进行分组，获取每个大科室里边的下级子科室
        Map<String, List<Department>> departmentMap =
                departmentList.stream().collect(Collectors.groupingBy(Department::getBigcode));
        for (Map.Entry<String, List<Department>> entry : departmentMap.entrySet()) {
            // 大科室编号
            String bigCode = entry.getKey();
            // 大科室编号对应的全部数据
            List<Department> departmentsList = entry.getValue();
            // 封装大科室
            DepartmentVo departmentVo = new DepartmentVo();
            departmentVo.setDepcode(bigCode);
            departmentVo.setDepname(departmentsList.get(0).getBigname());
            // 封装小科室
            List<DepartmentVo> children = new ArrayList<>();
            for (Department department : departmentsList) {
               DepartmentVo depVo =  new DepartmentVo();
               depVo.setDepcode(department.getDepcode());
               depVo.setDepname(department.getDepname());
               // 封装到list集合
               children.add(depVo);
            }
            // 把小科室list集合放到大科室children里面
            departmentVo.setChildren(children);
            // 最后放到最终的result里面
            result.add(departmentVo);
        }
        // 返回结果
        return result;
    }

    /**
     * 根据科室编号、医院编号，查询科室名称
     */
    @Override
    public String getDepName(String hoscode, String depcode) {
        Department department = departmentRepository.getDepartmentByHoscodeAndDepcode(hoscode, depcode);
        if (department != null) {
            return department.getDepname();
        }
        return null;
    }

    /**
     * 根据科室编号、医院编号，查询科室
     */
    @Override
    public Department getDepartment(String hoscode, String depcode) {
        return departmentRepository.getDepartmentByHoscodeAndDepcode(hoscode, depcode);
    }
}
